import os
import numpy as np
import tensorflow as tf
import input_data
import model

# you need to change the directories to yours.
train_dir = 'F:\\001-python\\train_1\\'
test_dir = 'F:\\001-python\\test_1\\'
train_logs_dir = 'F:\\001-python\\logs\\train\\'
val_logs_dir = 'F:\\001-python\\logs\\val\\'

from PIL import Image
import matplotlib.pyplot as plt

test = []
a=os.listdir(test_dir)
a.sort(key= lambda x:int(x[:-4]))
print(a)
for file in a:
    test.append(test_dir + file)
print('There are %d test pictures\n' % (len(test)))
print(test)
